"""
三角函数
"""

import torch


def test_tri():
    a = torch.rand(2, 3)
    print("==== torch.rand(2,3) ====")
    print(a)

    print("==== torch.cos(a) ====")
    print(torch.cos(a))
    print()

    a = torch.ones(2, 3)
    print("==== torch.ones(2,3) ====")
    print(a)

    print("==== torch.cos(a) ====")
    print(torch.cos(a))
    print()

    a = torch.zeros(2, 3)
    print("==== torch.zeros(2,3) ====")
    print(a)

    print("==== torch.cos(a) ====")
    print(torch.cos(a))
    print()


if __name__ == '__main__':
    test_tri()
